import os
import sys
from datetime import datetime

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from webdriver_manager.chrome import ChromeDriverManager


class Driver:
    driver = ""
    def __init__(self):
        options = webdriver.ChromeOptions()
        self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=options)
        self.driver.implicitly_wait(2)

    def getScressShot(self):
        #创建屏幕截图
        #图片文件名称：./2024-05-08-173456.png
        #../images/2024-05-08
        dirname = datetime.datetime.now().strftime("%Y-%m-%d")
        #判断dirname是否已近存在
        if not os.path.exists("images/" + dirname):
           os.mkdir("images/" + dirname)
        filename = sys._getframe().f_back.f_code.co_name + "-" + datetime.datetime.now().strftime("%Y-%m-%d-%H%M%S") + ".png"
        self.driver.save_screenshot("images/" + dirname + "/" + filename)

SearchDriver = Driver()